package cn.itcast.nio.c4;

import lombok.extern.slf4j.Slf4j;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import java.util.Set;

import static cn.itcast.nio.c2.ByteBufferUtil.debugAll;
import static cn.itcast.nio.c2.ByteBufferUtil.debugRead;

/**
 * 服务器端
 */
@Slf4j
public class Server {
    public static void main(String[] args) {
        m5();
    }

    /**
     * 4.4 处理 read 事件
     */
    private static void m5() {
        try (ServerSocketChannel ssc = ServerSocketChannel.open()) {
            ssc.bind(new InetSocketAddress(8080));
            log.info(ssc.toString());

            // 1.创建 selector，管理多个 channel
            Selector selector = Selector.open();
            ssc.configureBlocking(false);
            // 2.建立 selector 和 channel 的联系（注册）
            // SelectionKey 就是将来事件发生后，通过它可以知道事件和哪个 channel 的事件
            // key 只关注 accept 事件
            ssc.register(selector, SelectionKey.OP_ACCEPT);

            while (true) {
                // 3.select 方法，没有事件发生，线程阻塞，有事件，线程才会恢复运行
                // 事件发生后，要么处理，要么取消（cancel），不能什么都不做，否则下次该事件仍会触发，这是因为 nio 底层使用的是水平触发
                int count = selector.select();
//                int count = selector.selectNow();
                log.debug("select count: {}", count);
//                if(count <= 0) {
//                    continue;
//                }

                // 4.处理事件，selectedKeys 内部包含了所有发生的事件
                Set<SelectionKey> keys = selector.selectedKeys();

                // 遍历所有事件，逐一处理
                Iterator<SelectionKey> iter = keys.iterator();
                while (iter.hasNext()) {
                    SelectionKey key = iter.next();
                    log.debug("key：{}", key);
                    // 5.区分事件类型
                    if (key.isAcceptable()) { // 如果是 accept
                        ServerSocketChannel channel = (ServerSocketChannel) key.channel();
                        // 必须处理
                        SocketChannel sc = channel.accept();
                        sc.configureBlocking(false);
                        sc.register(selector, SelectionKey.OP_READ);
                        log.debug("连接已建立: {}", sc);
                    } else if (key.isReadable()) { // 如果是 read
                        try (SocketChannel sc = (SocketChannel) key.channel()) {
                            ByteBuffer buffer = ByteBuffer.allocate(128);
                            int read = sc.read(buffer);
                            if (read == -1) {
                                key.cancel();
                                sc.close();
                            } else {
                                buffer.flip();
                                debugAll(buffer);
                            }
                        }catch (IOException e){
                            e.printStackTrace();
                        }
                    }
                    // 处理完毕，必须将事件移除
                    iter.remove();
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 4.3 处理 accept 事件
     */
    private static void m3() {
        try (ServerSocketChannel ssc = ServerSocketChannel.open()) {
            ssc.bind(new InetSocketAddress(8080));
            log.info(ssc.toString());

            // 1.创建 selector，管理多个 channel
            Selector selector = Selector.open();
            ssc.configureBlocking(false);
            // 2.建立 selector 和 channel 的联系（注册）
            // SelectionKey 就是将来事件发生后，通过它可以知道事件和哪个 channel 的事件
            // key 只关注 accept 事件
            ssc.register(selector, SelectionKey.OP_ACCEPT);

            while (true) {
                // 3.select 方法，没有事件发生，线程阻塞，有事件，线程才会恢复运行
                // 事件发生后，要么处理，要么取消（cancel），不能什么都不做，否则下次该事件仍会触发，这是因为 nio 底层使用的是水平触发
                int count = selector.select();
//                int count = selector.selectNow();
                log.debug("select count: {}", count);
//                if(count <= 0) {
//                    continue;
//                }

                // 4.处理事件，selectedKeys 内部包含了所有发生的事件
                Set<SelectionKey> keys = selector.selectedKeys();

                // 遍历所有事件，逐一处理
                Iterator<SelectionKey> iter = keys.iterator();
                while (iter.hasNext()) {
                    SelectionKey key = iter.next();
                    log.debug("key：{}", key);
                    // 判断事件类型
                    if (key.isAcceptable()) {
                        ServerSocketChannel channel = (ServerSocketChannel) key.channel();
                        // 必须处理
                        SocketChannel sc = channel.accept();
                        log.debug("{}", sc);
                    }
                    // 处理完毕，必须将事件移除
                    iter.remove();
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 非阻塞模式
     */
    private static void m2() throws IOException {
        // 使用 nio 来理解非阻塞模式, 单线程
        // 0. ByteBuffer
        ByteBuffer buffer = ByteBuffer.allocate(16);

        // 1. 创建了服务器
        ServerSocketChannel ssc = ServerSocketChannel.open();
        ssc.configureBlocking(false); // 非阻塞模式
        // 2. 绑定监听端口
        ssc.bind(new InetSocketAddress(8080));
        // 3. 连接集合
        List<SocketChannel> channels = new ArrayList<>();
        while (true) {
            // 4. accept 建立与客户端连接， SocketChannel 用来与客户端之间通信
            SocketChannel sc = ssc.accept(); // 非阻塞，线程还会继续运行，如果没有连接建立，但sc是null
            if (sc != null) {
                log.debug("connected... {}", sc);
                sc.configureBlocking(false); // 非阻塞模式
                channels.add(sc);
            }
            for (SocketChannel channel : channels) {
                // 5. 接收客户端发送的数据
                int read = channel.read(buffer);// 非阻塞，线程仍然会继续运行，如果没有读到数据，read 返回 0
                if (read > 0) {
                    buffer.flip();
                    debugRead(buffer);
                    buffer.clear();
                    log.debug("after read...{}", channel);
                }
            }
        }
    }

    /**
     * 阻塞模式
     */
    private static void m1() throws IOException {
        // 使用 nio 来理解阻塞模式, 单线程

        // 0. ByteBuffer
        ByteBuffer buffer = ByteBuffer.allocate(16);
        // 1. 创建了服务器
        ServerSocketChannel ssc = ServerSocketChannel.open();
        ssc.configureBlocking(false); // 非阻塞模式

        // 2. 绑定监听端口
        ssc.bind(new InetSocketAddress(8080));

        // 3. 连接集合
        List<SocketChannel> channels = new ArrayList<>();
        while (true) {
            // 4. accept 建立与客户端连接， SocketChannel 用来与客户端之间通信
            log.debug("connecting...");
            SocketChannel sc = ssc.accept(); // 阻塞方法，线程停止运行
            log.debug("connected... {}", sc);
            channels.add(sc);
            for (SocketChannel channel : channels) {
                // 5. 接收客户端发送的数据
                log.debug("before read... {}", channel);
                channel.read(buffer); // 阻塞方法，线程停止运行
                buffer.flip();
                debugRead(buffer);
                buffer.clear();
                log.debug("after read...{}", channel);
            }
        }
    }
}
